import tushare as ts
from sqlalchemy import create_engine
import logging
from PledgeDetailData import save_history_data_pledge_detail
from PledgeDetailData import save_single_data_pledge_detail
import constant

"""
股票质押明细数据
"""
# 日志
logging.basicConfig(level=logging.DEBUG, format=' %(asctime)s - %(levelname)s - %(message)s')
logging.debug('------------------ Start of program ---------------------')

# 初始化数据库连接:
engine = create_engine(
    constant.get_db_path(),
    max_overflow=0,   # 超过连接池大小外最多创建的连接数
    pool_size=5,      # 连接池大小
    pool_timeout=30,  # 连接池中没有线程最多等待时间，否则报错
    pool_recycle=-1,  # 多久之后对连接池中的连接进行回收（重置）-1不回收
)

pro = ts.pro_api(constant.get_pro_token())

"""
由于源数据接口目前无法增量更新，本地更新的方法只能采取先比对数量，如果数量不等则删除，再重新插入数据 
"""
test_flag =False
if test_flag :
    ts_code = '000002.SZ'
    save_single_data_pledge_detail(ts_code, engine, pro)
else:
    save_history_data_pledge_detail(engine, pro)

logging.debug('------------------ end of program ---------------------')